Error Certificates for KV-Cache Eviction via Randomized Design
基于随机设计的 KV 缓存驱逐误差证书
🧭 随机化 KV cache 驱逐实现误差归因
Technical University of Munich
随机化 KV 缓存驱逐使系统能给出每步注意力误差的置信区间,用于归因缓存导致的失败
📌 工作参考价值(4 点)
- 在电商搜索/推荐/Agent 长上下文推理中,采用随机化逐出(如 Poisson 采样尾部 token)并计算误差证书,可在线检测 KV cache 压缩是否导致注意力偏差,触发全量重算或回退逻辑。
- 证书虽然预测失败的能力弱于输出 log-probability,但归因能力(AUC 0.73–0.75 vs. 0.47–0.54)可用于调度资源:仅对证书超阈值的高风险请求重新分配计算,避免浪费。
- 确定性 top-k 驱逐无法自我诊断(Theorem 1),因此生产系统中不应依赖保留状态计算的自检指标;改用随机化设计是从根本上获得可观测性。
- 工程实现上,可沿用论文的 Hájek 校正(单 logit 偏移)和调查抽样方差估计,无需大幅改动模型推理管线,即可获得每步无偏误差估计和 0.97 经验覆盖率。